package com.itheima.demo;

import java.io.File;
import java.util.HashMap;

public class Demo_File_002 {
    public static void main(String[] args) {
        File file = new File("demo_day10");
        HashMap<String, Integer> map = new HashMap<>();
        //创建一个方法;
        getCount(map, file);
        System.out.println(map);
    }

    private static void getCount(HashMap<String, Integer> map, File file) {
        //先进行判断传入的是否为null或者是否是一个文件;
        if (file == null || file.isFile()) {
            return;
        }
        //如果走到这里说明传入的不是null;就进入文件夹当中;
        File[] files = file.listFiles();
        for (File f : files) {
            if (f.isFile()) {
                //获取文件的名字;
                String fileName = f.getName();
                String[] split = fileName.split("\\.");
                if (split.length == 2) {
                    String fileEndName = split[1];
                    //进行判断;
                    if (map.containsKey(fileEndName)) {
                        Integer count = map.get(fileEndName);
                        count++;
                        map.put(fileEndName, count);
                    } else {
                        map.put(fileEndName, 1);
                    }
                }
            } else {
                getCount(map, f);
            }
        }
    }
}
